package cn.xxlucky.xssdk01.pojo.shard;

import com.alibaba.fastjson.annotation.JSONField;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;

@ApiModel("请求消耗的token数量实体类")
public class Usage {

    @ApiModelProperty(value = "用户输入token数量", example = "6")
    @JSONField(name = "prompt_tokens")
    private Integer promptTokens;

    @ApiModelProperty(value = "输出token数量", example = "42")
    @JSONField(name = "completion_tokens")
    private Integer completionTokens;

    @ApiModelProperty(value = "总token数量", example = "48")
    @JSONField(name = "total_tokens")
    private Integer totalTokens;

    public Integer getPromptTokens() {
        return promptTokens;
    }

    public void setPromptTokens(Integer promptTokens) {
        this.promptTokens = promptTokens;
    }

    public Integer getCompletionTokens() {
        return completionTokens;
    }

    public void setCompletionTokens(Integer completionTokens) {
        this.completionTokens = completionTokens;
    }

    public Integer getTotalTokens() {
        return totalTokens;
    }

    public void setTotalTokens(Integer totalTokens) {
        this.totalTokens = totalTokens;
    }
}
